package com.ruoyi.bearing.domain;

import com.ruoyi.common.annotation.Excel;
import com.ruoyi.common.core.domain.BaseEntity;

import java.util.Date;

public class ModelTrain extends BaseEntity {
    private static final long serialVersionUID = 1L;
    /**
     * 模型ID
     */
    @Excel(name = "模型ID")
    private String mid;

    public String getMid() {
        return mid;
    }

    public void setMid(String mid) {
        this.mid = mid;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public String getModel() {
        return model;
    }

    public void setModel(String model) {
        this.model = model;
    }

    public String getType() {
        return type;
    }

    public void setType(String type) {
        this.type = type;
    }

    public String getLr() {
        return lr;
    }

    public void setLr(String lr) {
        this.lr = lr;
    }

    public String getOptimizer() {
        return optimizer;
    }

    public void setOptimizer(String optimizer) {
        this.optimizer = optimizer;
    }

    public String getBatch() {
        return batch;
    }

    public void setBatch(String batch) {
        this.batch = batch;
    }
    public String getEpoch() {
        return epoch;
    }
    @Override
    public String toString() {
        return "ModelTrain{" +
                "mid='" + mid + '\'' +
                ", name='" + name + '\'' +
                ", model='" + model + '\'' +
                ", type='" + type + '\'' +
                ", lr='" + lr + '\'' +
                ", optimizer='" + optimizer + '\'' +
                ", batch='" + batch + '\'' +
                ", epoch='" + epoch + '\'' +
                ", layer='" + layer + '\'' +
                ", kernelSize='" + kernelSize + '\'' +
                ", c='" + c + '\'' +
                ", r='" + r + '\'' +
                ", timeStep='" + timeStep + '\'' +
                ", rnnUnits='" + rnnUnits + '\'' +
                ", createTime=" + createTime +
                '}';
    }

    public void setEpoch(String epoch) {
        this.epoch = epoch;
    }

    public String getLayer() {
        return layer;
    }

    public void setLayer(String layer) {
        this.layer = layer;
    }

    public String getKernelSize() {
        return kernelSize;
    }

    public void setKernelSize(String kernelSize) {
        this.kernelSize = kernelSize;
    }

    public String getC() {
        return c;
    }

    public void setC(String c) {
        this.c = c;
    }

    public String getR() {
        return r;
    }

    public void setR(String r) {
        this.r = r;
    }

    public String getTimeStep() {
        return timeStep;
    }

    public void setTimeStep(String timeStep) {
        this.timeStep = timeStep;
    }

    public String getRnnUnits() {
        return rnnUnits;
    }

    public void setRnnUnits(String rnnUnits) {
        this.rnnUnits = rnnUnits;
    }

    @Override
    public Date getCreateTime() {
        return createTime;
    }

    @Override
    public void setCreateTime(Date createTime) {
        this.createTime = createTime;
    }

    /**
     * 模型名称
     */
    @Excel(name = "模型名称")
    private String name;

    /**
     * 模型类型
     */
    private String model;

    /**
     * 模型类型：数据库
     */
    @Excel(name = "模型类型")
    private String type;
    /**
     * 学习率
     */
    @Excel(name = "学习率")
    private String lr;
    /**
     * 优化器
     */
    @Excel(name = "优化器")
    private String optimizer;
    /**
     * batch size
     */
    @Excel(name = "批次")
    private String batch;
    /**
     * 训练轮数
     */
    @Excel(name = "训练轮数")
    private String epoch;
    /**
     * 隐藏层数
     */
    @Excel(name = "隐藏层数")
    private String layer;
    /**
     * 卷积核大小
     */
    @Excel(name = "卷积核大小")
    private String kernelSize;
    /**
     * SVM C
     */
    @Excel(name = "惩罚系数")
    private String c;
    /**
     * SVM r
     */
    @Excel(name = "核系数")
    private String r;
    @Excel(name = "时间步长")
    private String timeStep;
    @Excel(name = "RNN Cel特征维数")
    private String rnnUnits;
    @Excel(name = "创建时间")
    private Date createTime;
}
